class Student:
    def __init__(self):
        self.stu_list = []
        self.dict_list = {}
    #新增好友的方法
    def add(self):
        add_friend = input("请输入要添加的好友：")
        self.stu_list.append(add_friend)
        print('好友添加成功')
    #修改好友的方法
    def update(self):
        if len(self.stu_list) == 0:
            print('好友列表为空')
        else:
            before_name = input("请输入要修改的好友姓名：")
            after_name = input("请输入修改后的好友姓名：")
            if before_name in self.stu_list:
                friend_index = self.stu_list.index(before_name)
                self.stu_list[friend_index] = after_name
                print("修改成功")
    #删除好友的方法
    def delete(self):
        if len(self.stu_list) != 0:
            del_name = input("请输入删除好友姓名：")
            # 删除的好友未分组
            if del_name in self.stu_list:
                self.stu_list.remove(del_name)
                print('删除成功')
    #查询好友的方法
    def list(self):
        view1 = input("是否显示分组的好友y/n\n")
        if(view1 =='y'):
            print("分组内容:",self.dict_list)
        else:
            print("好友列表:",self.stu_list)

    # 获取所有分组名称
    def get_all_groups_name(self):
        groups_name = []
        for dict_elem in self.dict_list:
                groups_name.append(dict_elem)
        return groups_name

    def group(self):
        create_group = input("是否创建新的分组y/n\n")
        friends = self.stu_list
        if create_group == 'y':
            if len(friends) == 0:
                print("当前没有任何好友")
            else:
                # 请创建分组
                group_name = input("请输入分组名称：\n")
                group_name_li = list()
                # 展示当前好友
                print(friends)
                # 移动联系人到哪个组
                friend_name = input("请输入好友名称：\n")
                if friend_name in friends:
                    all_friend = []
                    for friend_li_elem in self.stu_list:
                        all_friend.append(friend_li_elem)
                    if friend_name in all_friend:
                        group_name_li.append(friend_name)
                        self.stu_list.remove(friend_name)
                        # 构建字典： {组名称：分组列表}
                        friend_dict = dict()
                        friend_dict[group_name] = group_name_li
                        self.dict_list = friend_dict
                    else:
                        print("请输入正确的名称")
                else:
                    print('请输入正确好友名称')
        elif create_group == 'n':
            # 显示当前的分组，将用户添加到指定的组
            current_groups = self.dict_list
            print('当前分组:')
            for current_group in current_groups:
                    print(current_group)
            add_group = input('请选择添加的组:\n')
            # 判断用户输入的组名是否在当前已存在的分组名中
            if add_group in self.get_all_groups_name():
                # 添加好友到指定的组
                add_name = input('请选择添加的好友名称：\n')
                # 判断用户输入的好友是否存在好友列表中
                if add_name in self.stu_list:
                    # 判断用户是否在其他组中
                    if add_name not in current_groups:
                        # 将好友添加到指定的组内
                        self.stu_list.remove(add_name)
                        current_groups[add_group].append(add_name)
                        print('添加好友分组中')
                else:
                    print('该好友已在其他分组中')
            else:
                print('请输入正确的组名')
    def welcome(self):
        print("* * 欢迎使用好友管理系统 * * ")
        print("1：添加好友")
        print("2：删除好友")
        print("3：修改好友")
        print("4：展示好友")
        print("5：好友分组")
        print("6：退出")

        while True:
            option = input("请输入您需要的功能\n")
            # 添加好友
            if option == '1':
                self.add()
            elif option == '3':
                self.update()
            elif option == '2':
                self.delete()
            elif option=='4':
                self.list()
            elif option=='5':
                self.group()

if __name__ == '__main__':
    stu = Student()
    stu.welcome()
